package th.retrofit.lib;

/**
 * https://leetcode-cn.com/problems/sum-root-to-leaf-numbers/
 * 给你一个二叉树的根节点 root ，树中每个节点都存放有一个 0 到 9 之间的数字。
 * 每条从根节点到叶节点的路径都代表一个数字：
 *
 * 例如，从根节点到叶节点的路径 1 -> 2 -> 3 表示数字 123 。
 * 计算从根节点到叶节点生成的 所有数字之和 。
 *
 * 叶节点 是指没有子节点的节点。
 */
public class Solution129 {

    int sum = 0;
    public int sumNumbers(TreeNode root) {
        if (root == null) return sum;
        sumNumbers(root, 0);
        return sum;
    }

    private void sumNumbers(TreeNode root, int pre) {
        int num = pre;
        num = (num * 10) + root.val;
        if (root.left == null && root.right == null) {
            sum += num;
        }
        if (root.left != null) {
            sumNumbers(root.left, num);
        }
        if (root.right != null) {
            sumNumbers(root.right, num);
        }
    }

    public static void main(String[] args) {
        Integer[] arr = new Integer[]{4,9,0,5,1};
        TreeNode root = TreeNode.array2Tree(arr);
        int sum = new Solution129().sumNumbers(root);
        System.out.println(sum);
    }

}
